package com.duoduo.剑指Offer;

import com.duoduo.TreeNode;

import java.util.Stack;

/**
 * @program: algorithm
 * @description: 二叉树的镜像12
 * 请完成一个函数，输入一个二叉树，该函数输出它的镜像。
 *
 * 例如输入：
 *
 *      4
 *    /   \
 *   2     7
 *  / \   / \
 * 1   3 6   9
 * 镜像输出：
 *
 *      4
 *    /   \
 *   7     2
 *  / \   / \
 * 9   6 3   1
 *
 *  
 *
 * 示例 1：
 *
 * 输入：root = [4,2,7,1,3,6,9]
 * 输出：[4,7,2,9,6,3,1]
 *
 * @author: chaidl  ！
 * @create: 2022-05-23 23:00
 */
public class 二叉树的镜像12 {
	/**
	 * 采用递归的解法
	 * @param root
	 * @return
	 */
	public TreeNode mirrorTree(TreeNode root) {
		return dfs(root);
	}
	
	private TreeNode dfs(TreeNode root) {
		if (root == null){
			return null;
		}
		//先把根节点的左子树拿出来进行保存
		TreeNode tmp=root.left;
		//然后将根节点的左子树变为右子树，并且递归
		root.left=dfs(root.right);
		root.right = dfs(tmp);
		return root;
	}
	
	/**
	 * 采用栈
	 * @param root
	 * @return
	 */
	public TreeNode mirrorTree2(TreeNode root) {
		if (root == null) {
		    return null;
		}
		Stack<TreeNode> stack = new Stack<TreeNode>() ;
		//将根节点入栈
		stack.push(root);
		while(!stack.isEmpty()) {
			//出栈
			TreeNode node = stack.pop();
			//如果当前节点不为空，将他的左右子树加入栈
			if(node.left != null) stack.add(node.left);
			if(node.right != null) stack.add(node.right);
			TreeNode tmp = node.left;
			node.left = node.right;
			node.right = tmp;
		}
		return root;
	}
}
